#include <signal.h>
#include "../comm/httplib.h"
#include "oj_control.hpp"

using namespace httplib;
static Control *ctrl_ptr = nullptr;

// 收到信号全部上线
void Recovery(int signo)
{
    ctrl_ptr->RecoveryMachine();
}

int main()
{
    signal(SIGQUIT, Recovery);
    // 请求的路由功能
    /***********
     * 请求的分类
     *      获取所有题目列表
     *      根据题目编号，获取题目内容
     *      提交题目，判题
     */     

    Server svr;
    // 获取所有题目列表
    Control ctrl;
    ctrl_ptr = &ctrl;
    svr.Get("/all_questions", [&ctrl](const Request& req, Response& resp)
    {
        std::cout << "请求所有题目页面" << std::endl;
        std::string html;
        ctrl.AllQuestions(&html);
        resp.set_content(html, "text/html; charset=utf-8");
    });

    // 根据题目编号，获取题目内容
    // 正则表达式，d表示数字，+表示一个或多个代替
    svr.Get(R"(/question/(\d+))", [&ctrl](const Request& req, Response& resp)
    {
        std::cout << "请求单个题目页面" << std::endl;
        // 获取请求参数题目编号
        std::string html;
        std::string number = req.matches[1];
        ctrl.Question(number, &html);
        resp.set_content(html, "text/html; charset=utf-8");
    });
    // 提交题目，判题
    svr.Post(R"(/judge/(\d+))", [&ctrl](const Request& req, Response& resp)
    {
        std::cout << "请求判题\n";
        std::string out_json;
        std::string number = req.matches[1];
        ctrl.Judge(number, req.body, &out_json);
        // 获取请求参数题目编号
        resp.set_content(out_json, "application/json; charset=utf-8");
        std::cout << "等待用户提交中...\n";
    });

    svr.set_base_dir("./wwwroot");
    std::cout << "等待用户提交中...\n";
    svr.listen("0.0.0.0", 8000);
    return 0;
}